import pymysql
import streamlit as st
import function.database as dd
import time
import re
def init():
    st.set_page_config(layout="wide",
                       page_icon="😀",
                       page_title="聊天机器人")
    st.title("注册")
    return 0
def register():
    init()
    username=st.text_input("请输入用户名")
    password = st.text_input("请输入密码",type="password")
    repass = st.text_input("请再次输入密码", type="password")
    loginflag=st.button("已有帐号，点击登录!")
    registerflag=st.button("注册")
    if loginflag:
        st.switch_page("login.py")
    if registerflag:
        if username and password and repass:
            # 2、校验用户名的长度是否为11位 并且是否为手机号 正则表达式
            if re.match('^(13|15|17|18|19)[0-9]{9}$', username):
                # 3、查看两次密码是否一致 并且密码长度必须大于等于8位
                if password == repass and len(password) >= 8:
                    # 4、查询数据库是否有重复信息
                    if dd.query_user_no_exists(username):
                        dd.query_user(username)
                        st.success("注册成功")
                        time.sleep(2)
                        st.switch_page("login.py")
                    else:
                        st.error("用户已注册，请勿重复注册！")
                else:
                    st.error("两次密码不一致或者密码长度字段不足8位")
            else:
                st.error("手机号格式不正确")

        else:
            st.error("请务必填写相关注册信息")

register()
